Update src/main.py