Fixing documentation -> pass doc tests