Replace plot 4 code and graphic with better version