Add theme JS back to mkdocs