Added the ability to take user input