Changed column names in csv, so changed code to reflect this