Add tests for comments