Added plotting helper functions, returns and docstrings.