This code include the colab notebook used to generate figure 3.

Please install all libraries in your computer or upload the file to colab and run all the cells.
