Link to my notebook: https://colab.research.google.com/drive/1xCCkmpbQQJ-YrSu01U9ie6Gt2gtqnNNf?usp=sharing