add main.py to handle data