Hi T. Kipf, Thank you for the great code. Is it also possible for you to share the function you used to produce the figure you show in the paper:  Thank you!