Remove extra returns