Would be nice if inputs to functions could be agnostic about whether they were Numpy arrays or dataframes