Co-lab implementation: https://colab.research.google.com/drive/1jVtxlXHTnPqXn-7OlH3n5AfAri8wD2xB
202407091102