Incorporating some AI