Add functionality to remove comments.