Fix tests and update readme.