Now taking real training data, training and storing weights.