Minor improvements
Added more ai code
@