minor changes to use only 10 images for training and rest for testing