Updated neural net

First attempt at implementing training