Added checkboxes for AI