Updated main.cpp (Jonathan)