Fix docs

Slight doc fixes