Adding some test code