refactor(classification): add highlight when being edited