Refactor get_input - part II.