Add test case codes in comments

Add test case codes in comments