Add Unit test to cover more branches