Added slightly more ai behavior