Added some test code