@Nullable
        @Override
        protected PyClass getContainingClass(@NotNull PyFunction element) {
          return element.getContainingClass();
        }
@Nullable
        @Override
        protected PyClass getContainingClass(@NotNull PyFunction element)