Added more test - code coverage - 50%