Update docs
Tidy up