Image Classification using transfer learning with Resnet18