Testing AI code and basic math functions