Adding mathematica code