Refactor CommentsController