Tweak main.py.