Fix comment/docs