Add data loader option to keep input data