AIContext refactoring. AIContext is now a subclass of MethodContext