Adding basic test coverage