Added line to make sure that the user types a whole number