Pylearn updates