added neural net