ENH: added new plotting features