Update main.cpp

Changed the include