added neural net code