minor modifications to plotting stuff