Fixed normalization helper function. Norm over all data. Not over axis 0.