Improvements to the docs based on the last runthrough of the code