added some more test code