Artificial Neural Networks - ANN in Python - Step 2