PyLearn2 MLP updates for MNIST