Add some tests and fix Readme