Revert MaxIter & lambda, clean up back/forward prop code