Removed hardcoded path from mnist_variation.py