added some code on theano