Simple Mathematica code