small API change and doc update