Update theano_class.py