Optional import of plotting functions