Added a branch test
Added more unit tests brings coverage to 100%