Fixed main.cpp includes