Update src-python/main.py