Visual improvements to plotting widget