added target height as second input for Neural Network