minor ai improvements