Update docs

I was updating my code in line with latest updates and noticed the docs were not updated. This just keeps the docs up to date.