added some test code