main.py: minor style fixes