Adding tests to bump up code coverage