Documentation improvements
Minor bug fix.