changed probability calculations a little bit; added neural network stuff