Just a suggestion: ```getTimeStep()``` in stead of ```timestep()``` would make the code more readable (imho).