Before revision:  Our code is available at URL
After revision:  Google Colab walkthrough: URL