Add error handling on input.