Theano data generation now working