Add ImageGrid class. Output MNIST weights after training as png
cleanup