Complete test code