improve normalize function