improve docstring formatting and add picture to readme