Add includes to test.cpp