Adding some AI elements