Add some AI functionality ;)