Add tests, make things better
Add README.md