improved plots
(#343)