Improvments to the docs