Add tests
Full test coverage