Changing size of figure to improve plotting performance

Currently the size of figure is set very big which slow down the speed of plotting a lot. Size of figure should be set according what is needed to improve the performance