move main to main.py