added while loop to choice