Add test for comment