Add tests for coverage