Added first version of the neural net