Update README and add matplotlib_utils.py