Changed column names in csv, so changed code to match it