Jets rework Marshall's contract