Please fix issues with my return?