Add AI Mode Feature