Fix comments and add test.