Use return instead of echo to show results