updates, single iteration functions as intended, now adding multiple iterations and fixing scalibility issue of backprop