checkpoint: add main.py