Add docs
Several improvements