upgraded a few things in AI