using '# <codecell>' separator so the script is compatible with IPython notebook