Add comments, (very) minor changes