fix doc
improve code style
temp fix