fixed bug in diff equations for drive