Got AI wandering working better