Update docstrings and documentation

- add image
- fill README
- fix test