add option to save plot