Added a test at the end for user input