little update main.py