Update TRAIS script