added some new plotting features