Updating Module with SquaredDiffNode in Widget