add tests, update readme, add image