further updates to plotting method