Added code for the testing method using just a simple theano function