Getting the overfitting graphs right (earlier versions weren't initializing the weights correctly)