def add_hyperlink_to_notebook(notebook_path):
    """
    Add the colab link to notebook!
    """
    nb = nbformat.read(notebook_path, as_version=4)
    link_text = "https://colab.research.google.com/github/metadriverse/metadrive/blob/main/documentation/source/{}".format(
        os.path.basename(notebook_path))
    link_text = "[![Click and Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)]({})".format(
        link_text)
    # Assuming you want to add the link below the first heading
    cell = nb.cells[0]
    assert cell.cell_type == 'markdown', cell
    if "colab" not in cell.source:
        p = cell.source.find("\n")
        cell.source = cell.source[:p] + f"\n\n{link_text}\n" + cell.source[p:]
        # print(cell.source)
        nbformat.write(nb, notebook_path)

    Add the colab link to notebook!
    
documentation/source/conf.py