add time steps.
Need to refactor the code to generate csv