Update document and fix incorrect import