Changed the output of main.cpp