Add header to test problem

The test program seems to have been using the wrong time functions
and failing. Including time.h fixed the problem.