add saving and plotting options