Added plotting helper functions