Improve docstrings

Any more good suggestions @talumbau?