Added in test function to neural network class
Merged the mains