add main.py file
finish the main feature