differential equations code snippets in python