Results of my colab with @romysurjan