Add data_type argument