assignClasses function

Assign correct column classes