add setup.py and main function in main.py