Update plot4.py