Implement question 4c missing check_input