added some test code

Test code added.