added headers to test.cpp