Added function to remove comments