Update main.cpp

Reverted.