VGG and Transfer Learning - Code pt 2