python main function not returning the dataframe