Update maincode.py