lite modifieringar av ai-strukturen