Update main.py

Moved files and changed references in preparation of file-splitting.