Minor fixes to plotting methods