Better results with Theano