Added a freeze/unfreeze button for the plot.

Also added a couple comments and removed obsolete color code.