Update plot4.R

Tidy up