Updated the plots.
Minor bug fixes.