add ai part

add ai part