Tidy main.py slightly.