Told to add some test code