Update main.py

Règle tous les bugs présents jusqu'à maintenant.