Updated to remove 'puts' from return statement