Improve documentation and fix some typos