Cleaned up AI classes