Updating main.py to be a better example of functionality.