added some extra test code