Added timeStep implementation needs testing.