Really enjoyed your article. I was also able to simply just build on what you did. Really appreciate it.

Check it out here: https://colab.research.google.com/drive/1YfZJ0MGQUhdplI06uvtnUwmPJsIxkOYM