Small change to two plotting functions.