Update docs
Update docs (TODO: improve)