fix doc and code