Tweaks and docstrings
Add image to readme