added extra plotting functions