First stab at test code