Add some comments
Enforce ordering of the legend
Clean up the edge_weight sampling code; code to calculate expected edge_weight
Add transform_x method in order to plot log-scale
Remove pearson correlation
Allow for saving the plot to a file image