Improve plot docstring
Some small corrections to the documentation
Move visualization upwards in the index