forgot to add new AI classes