Assignment 4: Training and Testing with Keras