Added initial working neural network