Ex4 NN - Figure out cost function