Add correct icon class