add method to test on the test set, print some extra stuff out for now