Add basic test coverage