update plotting
minor fix