added support for mnist-variations