Plot 2 more update

More updates to the plot 2 script