Add tests
Update README