Minor changes toward implementing neural network