adding some of hte neural net code