Updates to the plotting functions.