Add ai play set logic