fix a bug
add docs