Update main.py

update develop branch from website.