Write some test code