working on timesteps. Will finish it later