Removing unneeded variable from return