Done with theano matrices