Reshaping and normalizing

[balaji.ambresh]

Make use of images parameter consistently (i.e. don’t use training_images). In your code, the reshaped array is lost due to incorrect assignments.