Adding unit tests to get code coverage up