Unfortunately, Mito only works in JupyterLab and Jupyter notebooks right now. Support for VSCode and Google Colab is on the roadmap.