Fix documentation

As for Issue #4