Implement user input when testing