Improve docs
Slight code fixes