Errors and mismatches: https://colab.research.google.com/drive/1m56uhwHsf1bkyIymW2hBWEsdmzhpIKC2?usp=sharing#scrollTo=41x859g11YOD