Update main.cpp

update the code in my phone