Add and remove necessary classes on control update.