(minor changes to plotting)