Updated from forked file. Completed section 1 of the exercise, allowing for user input.