generator.py: Fixed formatting