small change to plotting