Add controller registration