Played with the Mathematica notebook

Updated solution section to paper