add some test code