Add class to editor content to show that a unit is selected