added neural net data gen