improve plots and other things