Edited main.cpp, added some includes which will be needed