I changed to a vectorized method for back prop
Implementation with a for loop for back prop. It is still not working