Add normalize function