Return statement updated