Finally got the reg gradient decent right!