add class for correct style