Corrected main.cpp again :P