Adds WIP for matplotlib plot support