Update main.cpp

bugs correction