add in class for columns