Remove unnecessary headers in main.cpp