differential equation in python