Enhancement for plot3 and plot4

Enhancement for plot3 and plot4