Added stuff to the test file