Small fix and documentation update