Add some logic to determine the comment type