Optimise function return