Improved plots
Better captions