Increased the number of outputs
Added correct time step for saving