if anyone would like to colab with me. https://docs.google.com/document/d/1Ql8NKvJ8e0FzxOpzqZ9lc1nay72GW9TZSKqsiG5xRAs/edit?usp=sharing