Add class depending on the current action in controls