feat(tests): Added tests to reach desired test coverage levels