Updated broken main.cpp file