add test case modified and address some comments