Add tests to 100%