nice comments
update test to work with the new changes in theano