It is recommended that this be done in a separate .py file and use the main submission as an import.