changing print to return