add main.py for example