Update main.cpp (testing 1.1)