Add missed css classes