Add README images and update plotting.py