Improved and corrected in-code docs