Thanks Alisher, yes now no need to define a separate function to get the column names and change numpy array to dataframe :)