Minor fixes for plot4

Minor fixes for plot4