add right classes