Neural network 1st assignment