Small fix to logging, improved docs