fixed AI to work more like intended