Added basic tests to get code coverage