Fixing documentation and very small improvement