added neural network prediction