Now we can restart a snapshot
Tested all functionalities in this file using mock XOR data
TODO: No validation done during training. Split up training data into test + training. Use portion of training data as validation data to measure ACC during training.