Add tests
Improve readme