Added ability for user input