updated Mathematica notebook and added python plot file for results