minor fixes to output plot