Did some experiments. Nailed 3-layer MLP for MNIST. Adding the code.