fixed initial condition
minor fixes in timestep