Fix main.cpp by rebasing