Add test cases to ex02