updated plotting
updated plotting