added skaae's version of dataset-mnist.lua