main.py: Modified file generation