Update docs
Update tests