revise column names based on week 4 lectures