Added some mathematica code