Added lines to allow for headless operation of tests.