Add begin-rescue for invalid user input