Final cleanup

Fixed some comments, tested, figured out problem where no box around
legend obliterated part of the box around the plot for plot 4