Add function to remove comments.