Adding more tests to increase code coverage