Fix some of the docs