update sample main.py