Cleaned MNIST code - added deeper layer