Re-add ai script