Allow user to do x^2 along with square(x). Rename linear to affine.