Adding neural net code