Added plot and 3d plot