Refactor dataset comments API