Move main program into main.py