fix comments, add to test