A few more minor fixes after the build cmd landed

Some flaky test improvements as well as some updates to the styling of the docs.