Python State Space Equations