diff --git a/README.md b/README.md
index 78667e4..5bbef68 100644
--- a/README.md
+++ b/README.md
@@ -66,6 +66,14 @@ pdd --include=src/**/*.py # include only .py files in src/
 pdd --include=src/**/* # include all files in src/
 ```
 
+`` --model ``
+
+Opt-in to use inbuilt ML model to prioritize puzzles found in code. 
+```bash
+pdd --model
+pdd -m
+```
+
 ## How to Format?
 
 Every puzzle has to be formatted like this (pay attention
diff --git a/assets/puzzles.xsd b/assets/puzzles.xsd
index 5551b72..afa5413 100644
--- a/assets/puzzles.xsd
+++ b/assets/puzzles.xsd
@@ -98,6 +98,7 @@ SOFTWARE.
       </xs:sequence>
       <xs:attribute name="version" use="required" type="xs:string"/>
       <xs:attribute name="date" use="required" type="xs:dateTime"/>
+      <xs:attribute name="model" type="xs:boolean"/>
     </xs:complexType>
     <xs:unique name="puzzleBody">
       <xs:selector xpath="./puzzle"/>
diff --git a/bin/pdd b/bin/pdd
index 899cc39..5564b4d 100755
--- a/bin/pdd
+++ b/bin/pdd
@@ -68,6 +68,7 @@ begin
         '-r', '--rule', 'Rule to apply (can be used many times)',
         delimiter: ';'
       )
+      o.bool '-m', '--model', 'Opt-in to use ml model to prioritize puzzles'
     end
   rescue Slop::Error => e
     raise StandardError, "#{e.message}, try --help"
diff --git a/lib/pdd.rb b/lib/pdd.rb
index 03633fc..72c9608 100644
--- a/lib/pdd.rb
+++ b/lib/pdd.rb
@@ -121,7 +121,7 @@ def attrs
         'xsi:noNamespaceSchemaLocation' => "#{host('xsd')}/#{PDD::VERSION}.xsd",
         'version' => PDD::VERSION,
         'date' => Time.now.utc.iso8601
-      }
+      }.merge(@opts[:model] ? { 'model' => true } : {})
     end
 
     def host(suffix)