Complete test code and comment