enh: option to keep code