Add test for remove comments