added test cases and made code improvements #15

Closed owaisrazaeng1 closed 1 year ago