Added initial work on cifar10 data, saving and loading model