cleaned up AI for Josh