add option for plot type selection