added code to calculate time
timesteps work