adds some code for Neural Network regression