Added a booelan to decide whether to normalize after each timestep.
Added a booelan to decide whether to normalize after each timestep.