add check_input function