improve test code with comments