Neural Network from scratch-part 2