Add possibility to use all data for training(zero for testing)
Remove unnecessary writing temporary grayscale images