Working on Neural Network some more
Merging some work