added the plotting
updated