Remove unneccessary code in main.cpp