Add controller decorator