Taken https://colab.research.google.com/drive/1MSIkZy671MyEMyTK3HTBYAuOdzwrgt8Q as base

Original writeup (https://colab.research.google.com/drive/1P6VZeaG43v6Hc_33kWxM5jMThl3BkBkn).