Adds unit test for comments