Home work 1 (NN training)