add ai control