Fixing an error in a loop regarding user input. (TB)
fixed a mistake of mine. (TB)