Changed feedforward to return vector instead of index of highest value.
 - This will be used in training.
 
Wrote a seperate function to return the index of the highest value
 - This will be used in testing.