Return statement works