My code and guide code in the following link: https://colab.research.google.com/drive/103_aV-3e8mGLQeokcIUFFxd1Jwy2Hv1E?usp=sharing#scrollTo=TfxwAzFH4k_j