Update plotting.py