Update mainfile.py