updated the code to match the latest extension system for mkdocs