First attempt at traing Neural Net (added data)