added neural net module