Update main.cpp

fixed bug