added enhance_plot method
forgot to do that in an earlier commit