add docs
fix test