Fix issues with documentation