updated plotting with shading