Update and fix docs