add tow more test case and got a 100% coverage