More updates and improvements to the docs