Adds UI to main

Adds a "user interface" to the program, rather than hard coding in
values into main.