Adding code to remove comments