Fix some of the documentation