Update to main.py and data