Add controller hook functionality