diff --git a/src/index.tsx b/src/index.tsx
index f895c2c..3203a6b 100644
--- a/src/index.tsx
+++ b/src/index.tsx
@@ -138,6 +138,7 @@ export type FilterInput = {
   cueBallSpeed?: InputMaybe<RangeFilter>;
   cueObjectAngle?: InputMaybe<RangeFilter>;
   cueObjectDistance?: InputMaybe<RangeFilter>;
+  difficulty?: InputMaybe<RangeFilter>;
   intendedPocketType?: InputMaybe<Array<PocketEnum>>;
   make?: InputMaybe<Array<Scalars["Boolean"]["input"]>>;
   orFilters?: InputMaybe<OrFilter>;
@@ -262,6 +263,7 @@ export type PocketPointsGql = {
 
 export type PocketingIntentionFeaturesGql = {
   __typename?: "PocketingIntentionFeaturesGQL";
+  difficulty?: Maybe<Scalars["Float"]["output"]>;
   intendedPocketType?: Maybe<PocketEnum>;
   make?: Maybe<Scalars["Boolean"]["output"]>;
   targetPocketDistance?: Maybe<Scalars["Float"]["output"]>;
@@ -354,6 +356,7 @@ export type StreamErrorGql = {
 
 export type TargetMetricsGql = {
   __typename?: "TargetMetricsGQL";
+  averageDifficulty?: Maybe<Scalars["Float"]["output"]>;
   count: Scalars["Int"]["output"];
   makePercentage?: Maybe<Scalars["Float"]["output"]>;
 };
diff --git a/src/schema.gql b/src/schema.gql
index 9b9c9c3..7936c60 100644
--- a/src/schema.gql
+++ b/src/schema.gql
@@ -35,6 +35,7 @@ type AggregationIdentifierGQL {
 type TargetMetricsGQL {
   count: Int!
   makePercentage: Float
+  averageDifficulty: Float
 }
 
 input AggregateInputGQL {
@@ -68,6 +69,7 @@ input FilterInput {
   targetPocketDistance: RangeFilter = null
   cueObjectAngle: RangeFilter = null
   cueBallSpeed: RangeFilter = null
+  difficulty: RangeFilter = null
   intendedPocketType: [PocketEnum!] = null
   shotDirection: [ShotDirectionEnum!] = null
   videoId: [Int!] = null
@@ -161,6 +163,7 @@ type PocketingIntentionFeaturesGQL {
   targetPocketDistance: Float
   make: Boolean
   intendedPocketType: PocketEnum
+  difficulty: Float
 }
 
 type BankFeaturesGQL {