Update main.cpp

Amélioration du programme de base.