add tests
update image in readme