Code for Pylearn2