Clean Up

Cleaned up the code and created separate files for the classes and the
main loop. Running main.py sets up the environment and pulls the classes
from the classes.py file.