Update main.py

Version 1. Not the current version.