Update main.py

peut-être plus joli ^^