Fixed code so that dataframe is float instead of strings