fix identity handling

Before, using this

A = as_matrix([[x[i]*x[j] for j in range(2)] for i in range(2)]) + I

in a ufl file didn't compile, but

A = as_matrix([[x[i]*x[j] + I[i,j] for j in range(2)] for i in range(2)])

did.