Added while loop for selection Yes and no