forward pass and error calculation working FINALLY