Add a normalise function on return