instead of having two methods, return dataframe as tuple before and after creating column. and access by index.