Add more tests
More code coverage added