Mnist tweaks