Add integration test for comments