Add tests and update readme