Added initial data set
Trying to train data properly
added 2 hidden layers