Would be good to deal with data shape information
Would be good to deal with data shape information