Updated AI to use refactored method names