Add R code for plotting optimized node selection probabilities