Documentation improvements and new functionality