Add test cases to ex01