Return instead of echo