Update main.py

Version Beta 1