This should almost certainly just be done in main.py.