Fix main.cpp