Improve documentation

Cleanup some existing docs and extend the FSMLockMixin docs.