Add controller for comments