Update last part of TEDI assignment