Fix arrays names when doing arithmetic

Fixes #254.
Merge pull request #257 from shoyer/fix-math-name

Fix arrays names when doing arithmetic