Changed time_average to work with input with only one timestep (assuming it is already an average)