Import Data, Define Model (Image Classification with Pytorch - Part 1)