Add some test code for fun