some AI improvement