bug fix
updated docs