Add tests, and coverage is 100%