return in while loop