Python Differential Equations