Minor modification on plot figure