Updates comments for docs