cleaned up the MNIST demo code a bit - with 2 layers of [1024] units, can get up to 98.96% accuracy! that matches what's being reported in the literature
fixed some merge issues