diff --git a/.gitignore b/.gitignore
index 373edb9..156481d 100644
--- a/.gitignore
+++ b/.gitignore
@@ -37,3 +37,5 @@ next-env.d.ts
 
 # ignored for the template, you probably want to remove it:
 package-lock.json
+
+.env.local
diff --git a/app/play/[level]/page.tsx b/app/play/[level]/page.tsx
index 0c4f886..59f3918 100644
--- a/app/play/[level]/page.tsx
+++ b/app/play/[level]/page.tsx
@@ -39,6 +39,8 @@ export default function PlayLevelPage({
   const [placementMode, setPlacementMode] = useState<PlacementMode>("player");
   const [blockCount, setBlockCount] = useState(0);
   const [landmineCount, setLandmineCount] = useState(0);
+  const [maxBlocks, setMaxBlocks] = useState(0);
+  const [maxLandmines, setMaxLandmines] = useState(0);
   const flags = useQuery(api.flags.getFlags);
   const [mode, setMode] = useState<"play" | "test">("play");
   const [replaySpeed, setReplaySpeed] = useState(DEFAULT_REPLAY_SPEED);
@@ -47,6 +49,8 @@ export default function PlayLevelPage({
   useEffect(() => {
     if (map) {
       setPlayerMap(map.grid.map((row) => [...row]));
+      setMaxBlocks(Number(map.maxBlocks) ?? 0);
+      setMaxLandmines(Number(map.maxLandmines) ?? 0);
     }
   }, [map]);
 
@@ -219,20 +223,27 @@ export default function PlayLevelPage({
                 >
                   Place Player
                 </Button>
-                <Button
-                  onClick={() => handlePlacementModeChange("block")}
-                  variant={placementMode === "block" ? "default" : "outline"}
-                  className="h-10"
-                >
-                  Place Block ({2 - blockCount} left)
-                </Button>
-                <Button
-                  onClick={() => handlePlacementModeChange("landmine")}
-                  variant={placementMode === "landmine" ? "default" : "outline"}
-                  className="h-10"
-                >
-                  Place Landmine ({1 - landmineCount} left)
-                </Button>
+                {maxBlocks > 0 && (
+                  <Button
+                    onClick={() => handlePlacementModeChange("block")}
+                    variant={placementMode === "block" ? "default" : "outline"}
+                    className="h-10"
+                  >
+                    Place Block ({maxBlocks - blockCount} left)
+                  </Button>
+                )}
+
+                {maxLandmines > 0 && (
+                  <Button
+                    onClick={() => handlePlacementModeChange("landmine")}
+                    variant={
+                      placementMode === "landmine" ? "default" : "outline"
+                    }
+                    className="h-10"
+                  >
+                    Place Landmine ({maxLandmines - landmineCount} left)
+                  </Button>
+                )}
               </div>
               <div className="mb-8 flex flex-col items-center">
                 <h2 className="mb-4 text-xl font-semibold">
diff --git a/app/playground/page.tsx b/app/playground/page.tsx
index f55bd46..437b043 100644
--- a/app/playground/page.tsx
+++ b/app/playground/page.tsx
@@ -26,6 +26,8 @@ import {
   DialogHeader,
   DialogTitle,
 } from "@/components/ui/dialog";
+import { Input } from "@/components/ui/input";
+import { Label } from "@/components/ui/label";
 import { useToast } from "@/components/ui/use-toast";
 import { api } from "@/convex/_generated/api";
 import { Id } from "@/convex/_generated/dataModel";
@@ -61,6 +63,8 @@ export default function PlaygroundPage() {
   const [userSolution, setUserSolution] = useState<string[][]>([]);
   const [visualizingUserSolution, setVisualizingUserSolution] = useState(false);
   const [openSignInModal, setOpenSignInModal] = useState(false);
+  const [maxBlocks, setMaxBlocks] = useState(2);
+  const [maxLandmines, setMaxLandmines] = useState(0);
 
   const {
     isSimulating,
@@ -89,7 +93,7 @@ export default function PlaygroundPage() {
     setPublishing(true);
 
     try {
-      const submitted = await submitMap({ map });
+      const submitted = await submitMap({ map, maxBlocks, maxLandmines });
 
       if (submitted == 200) {
         toast({
@@ -341,9 +345,9 @@ export default function PlaygroundPage() {
                                 if (cell === " ") {
                                   if (playerCount === 0) {
                                     newMap[y][x] = "P";
-                                  } else if (blockCount < 2) {
+                                  } else if (blockCount < maxBlocks) {
                                     newMap[y][x] = "B";
-                                  } else if (landmineCount === 0) {
+                                  } else if (landmineCount === maxLandmines) {
                                     newMap[y][x] = "L";
                                   }
                                 } else if (cell === "P") {
@@ -472,6 +476,31 @@ export default function PlaygroundPage() {
                 </>
               ) : (
                 <>
+                  <div className="flex gap-2">
+                    <div>
+                      <Label htmlFor="block">Max Block</Label>
+                      <Input
+                        placeholder="Max Blocks"
+                        type="number"
+                        id="block"
+                        onChange={(e) => setMaxBlocks(Number(e.target.value))}
+                        value={maxBlocks}
+                      />
+                    </div>
+                    <div>
+                      <Label htmlFor="landmine">Max Landmine</Label>
+                      <Input
+                        placeholder="Max Landmines"
+                        type="number"
+                        id="landmine"
+                        onChange={(e) =>
+                          setMaxLandmines(Number(e.target.value))
+                        }
+                        value={maxLandmines}
+                      />
+                    </div>
+                  </div>
+
                   <div className="flex flex-col">
                     <Button
                       disabled={publishing}
diff --git a/components/ui/label.tsx b/components/ui/label.tsx
new file mode 100644
index 0000000..5341821
--- /dev/null
+++ b/components/ui/label.tsx
@@ -0,0 +1,26 @@
+"use client"
+
+import * as React from "react"
+import * as LabelPrimitive from "@radix-ui/react-label"
+import { cva, type VariantProps } from "class-variance-authority"
+
+import { cn } from "@/lib/utils"
+
+const labelVariants = cva(
+  "text-sm font-medium leading-none peer-disabled:cursor-not-allowed peer-disabled:opacity-70"
+)
+
+const Label = React.forwardRef<
+  React.ElementRef<typeof LabelPrimitive.Root>,
+  React.ComponentPropsWithoutRef<typeof LabelPrimitive.Root> &
+    VariantProps<typeof labelVariants>
+>(({ className, ...props }, ref) => (
+  <LabelPrimitive.Root
+    ref={ref}
+    className={cn(labelVariants(), className)}
+    {...props}
+  />
+))
+Label.displayName = LabelPrimitive.Root.displayName
+
+export { Label }
diff --git a/convex/README.md b/convex/README.md
new file mode 100644
index 0000000..4d82e13
--- /dev/null
+++ b/convex/README.md
@@ -0,0 +1,90 @@
+# Welcome to your Convex functions directory!
+
+Write your Convex functions here.
+See https://docs.convex.dev/functions for more.
+
+A query function that takes two arguments looks like:
+
+```ts
+// functions.js
+import { query } from "./_generated/server";
+import { v } from "convex/values";
+
+export const myQueryFunction = query({
+  // Validators for arguments.
+  args: {
+    first: v.number(),
+    second: v.string(),
+  },
+
+  // Function implementation.
+  handler: async (ctx, args) => {
+    // Read the database as many times as you need here.
+    // See https://docs.convex.dev/database/reading-data.
+    const documents = await ctx.db.query("tablename").collect();
+
+    // Arguments passed from the client are properties of the args object.
+    console.log(args.first, args.second);
+
+    // Write arbitrary JavaScript here: filter, aggregate, build derived data,
+    // remove non-public properties, or create new objects.
+    return documents;
+  },
+});
+```
+
+Using this query function in a React component looks like:
+
+```ts
+const data = useQuery(api.functions.myQueryFunction, {
+  first: 10,
+  second: "hello",
+});
+```
+
+A mutation function looks like:
+
+```ts
+// functions.js
+import { mutation } from "./_generated/server";
+import { v } from "convex/values";
+
+export const myMutationFunction = mutation({
+  // Validators for arguments.
+  args: {
+    first: v.string(),
+    second: v.string(),
+  },
+
+  // Function implementation.
+  handler: async (ctx, args) => {
+    // Insert or modify documents in the database here.
+    // Mutations can also read from the database like queries.
+    // See https://docs.convex.dev/database/writing-data.
+    const message = { body: args.first, author: args.second };
+    const id = await ctx.db.insert("messages", message);
+
+    // Optionally, return a value from your mutation.
+    return await ctx.db.get(id);
+  },
+});
+```
+
+Using this mutation function in a React component looks like:
+
+```ts
+const mutation = useMutation(api.functions.myMutationFunction);
+function handleButtonPress() {
+  // fire and forget, the most common way to use mutations
+  mutation({ first: "Hello!", second: "me" });
+  // OR
+  // use the result once the mutation has completed
+  mutation({ first: "Hello!", second: "me" }).then((result) =>
+    console.log(result),
+  );
+}
+```
+
+Use the Convex CLI to push your functions to a deployment. See everything
+the Convex CLI can do by running `npx convex -h` in your project root
+directory. To learn more, launch the docs with `npx convex docs`.
diff --git a/convex/_generated/api.d.ts b/convex/_generated/api.d.ts
index 719e27d..7ba585b 100644
--- a/convex/_generated/api.d.ts
+++ b/convex/_generated/api.d.ts
@@ -1,5 +1,3 @@
-/* prettier-ignore-start */
-
 /* eslint-disable */
 /**
  * Generated `api` utility.
@@ -153,5 +151,3 @@ export declare const components: {
     };
   };
 };
-
-/* prettier-ignore-end */
diff --git a/convex/_generated/api.js b/convex/_generated/api.js
index 853a45b..44bf985 100644
--- a/convex/_generated/api.js
+++ b/convex/_generated/api.js
@@ -1,5 +1,3 @@
-/* prettier-ignore-start */
-
 /* eslint-disable */
 /**
  * Generated `api` utility.
@@ -23,5 +21,3 @@ import { anyApi, componentsGeneric } from "convex/server";
 export const api = anyApi;
 export const internal = anyApi;
 export const components = componentsGeneric();
-
-/* prettier-ignore-end */
diff --git a/convex/_generated/dataModel.d.ts b/convex/_generated/dataModel.d.ts
index 91445df..8541f31 100644
--- a/convex/_generated/dataModel.d.ts
+++ b/convex/_generated/dataModel.d.ts
@@ -1,5 +1,3 @@
-/* prettier-ignore-start */
-
 /* eslint-disable */
 /**
  * Generated data model types.
@@ -60,5 +58,3 @@ export type Id<TableName extends TableNames | SystemTableNames> =
  * `mutationGeneric` to make them type-safe.
  */
 export type DataModel = DataModelFromSchemaDefinition<typeof schema>;
-
-/* prettier-ignore-end */
diff --git a/convex/_generated/server.d.ts b/convex/_generated/server.d.ts
index 788e90c..b5c6828 100644
--- a/convex/_generated/server.d.ts
+++ b/convex/_generated/server.d.ts
@@ -1,5 +1,3 @@
-/* prettier-ignore-start */
-
 /* eslint-disable */
 /**
  * Generated utilities for implementing server-side Convex query and mutation functions.
@@ -149,5 +147,3 @@ export type DatabaseReader = GenericDatabaseReader<DataModel>;
  * for the guarantees Convex provides your functions.
  */
 export type DatabaseWriter = GenericDatabaseWriter<DataModel>;
-
-/* prettier-ignore-end */
diff --git a/convex/_generated/server.js b/convex/_generated/server.js
index 1129936..4a21df4 100644
--- a/convex/_generated/server.js
+++ b/convex/_generated/server.js
@@ -1,5 +1,3 @@
-/* prettier-ignore-start */
-
 /* eslint-disable */
 /**
  * Generated utilities for implementing server-side Convex query and mutation functions.
@@ -90,5 +88,3 @@ export const internalAction = internalActionGeneric;
  * @returns The wrapped endpoint function. Route a URL path to this function in `convex/http.js`.
  */
 export const httpAction = httpActionGeneric;
-
-/* prettier-ignore-end */
diff --git a/convex/maps.ts b/convex/maps.ts
index 45e9288..8787e0a 100644
--- a/convex/maps.ts
+++ b/convex/maps.ts
@@ -159,6 +159,8 @@ export const addMap = mutation({
 export const submitMap = authenticatedMutation({
   args: {
     map: v.array(v.array(v.string())),
+    maxBlocks: v.number(),
+    maxLandmines: v.number(),
   },
   handler: async (ctx, args) => {
     try {
@@ -170,6 +172,8 @@ export const submitMap = authenticatedMutation({
         await ctx.db.insert("maps", {
           grid: args.map,
           level: undefined,
+          maxBlocks: args.maxBlocks,
+          maxLandmines: args.maxLandmines,
           submittedBy: ctx.userId,
           isReviewed: false,
         });
@@ -193,11 +197,15 @@ export const seedMaps = internalMutation({
         if (existingMap) {
           ctx.db.patch(existingMap._id, {
             grid: map.grid,
+            maxBlocks: 2,
+            maxLandmines: 0,
           });
         } else {
           ctx.db.insert("maps", {
             level: idx + 1,
             grid: map.grid,
+            maxBlocks: 2,
+            maxLandmines: 0,
             isReviewed: true,
           });
         }
@@ -362,6 +370,7 @@ export const playMapAction = internalAction({
       args.modelId,
       map.grid,
       activePrompt.prompt,
+      map.maxBlocks,
     );
 
     await ctx.runMutation(internal.results.updateResult, {
@@ -423,7 +432,12 @@ export const testAIModel = action({
       outputTokens,
       totalTokensUsed,
       totalRunCost,
-    } = await runModel(args.modelId, map.grid, activePrompt.prompt);
+    } = await runModel(
+      args.modelId,
+      map.grid,
+      activePrompt.prompt,
+      map.maxBlocks,
+    );
 
     return {
       map: solution,
diff --git a/convex/prompts.ts b/convex/prompts.ts
index 40225de..21a3749 100644
--- a/convex/prompts.ts
+++ b/convex/prompts.ts
@@ -12,6 +12,7 @@ const defaultPrompt = `Your task is to play a game.  We will give you a 2d array
 - You can not place the player in a location already used by a zombie or rock.
 - You can not place a block over the player or another block.
 - You must place both blocks and the player before starting the game.
+- You can place {MAX_BLOCKS} blocks on the map.
 
 # Grid Descriptions
 The 2d Grid is made up of characters, where each character has a meaning.
diff --git a/convex/schema.ts b/convex/schema.ts
index c4addf4..98ab790 100644
--- a/convex/schema.ts
+++ b/convex/schema.ts
@@ -15,6 +15,8 @@ export default defineSchema({
   maps: defineTable({
     level: v.optional(v.number()),
     grid: v.array(v.array(v.string())),
+    maxBlocks: v.optional(v.number()),
+    maxLandmines: v.optional(v.number()),
     submittedBy: v.optional(v.id("users")),
     isReviewed: v.boolean(),
   })
diff --git a/models/index.ts b/models/index.ts
index 927d674..ee24ce4 100644
--- a/models/index.ts
+++ b/models/index.ts
@@ -52,6 +52,7 @@ export async function runModel(
   modelId: string,
   map: string[][],
   prompt: string,
+  maxBlocks?: number,
   retry = 1,
 ): Promise<RunModelResult> {
   const userPrompt =
@@ -60,27 +61,31 @@ export async function runModel(
 
   let result;
   let reasoning: string | null = null;
+  const modifiedPrompt = prompt.replace(
+    "{MAX_BLOCKS}",
+    maxBlocks?.toString() ?? "",
+  );
 
   try {
     switch (modelId) {
       case AI_MODELS["gemini-1.5-pro"].slug: {
-        result = await gemini15pro(prompt, userPrompt, CONFIG);
+        result = await gemini15pro(modifiedPrompt, userPrompt, CONFIG);
         break;
       }
       case AI_MODELS["gpt-4o"].slug: {
-        result = await gpt4o(prompt, userPrompt, CONFIG);
+        result = await gpt4o(modifiedPrompt, userPrompt, CONFIG);
         break;
       }
       case AI_MODELS["claude-3.5-sonnet"].slug: {
-        result = await claude35sonnet(prompt, userPrompt, CONFIG);
+        result = await claude35sonnet(modifiedPrompt, userPrompt, CONFIG);
         break;
       }
       case AI_MODELS["perplexity-llama-3.1"].slug: {
-        result = await perplexityLlama31(prompt, userPrompt, CONFIG);
+        result = await perplexityLlama31(modifiedPrompt, userPrompt, CONFIG);
         break;
       }
       case AI_MODELS["mistral-large-2"].slug: {
-        result = await mistralLarge2(prompt, userPrompt, CONFIG);
+        result = await mistralLarge2(modifiedPrompt, userPrompt, CONFIG);
         break;
       }
       default: {
@@ -125,6 +130,6 @@ export async function runModel(
       };
     }
 
-    return await runModel(modelId, map, prompt, retry + 1);
+    return await runModel(modelId, map, modifiedPrompt, maxBlocks, retry + 1);
   }
 }
diff --git a/package.json b/package.json
index ea4f0eb..45c41e9 100644
--- a/package.json
+++ b/package.json
@@ -24,6 +24,7 @@
     "@radix-ui/react-dialog": "^1.1.2",
     "@radix-ui/react-dropdown-menu": "^2.1.1",
     "@radix-ui/react-icons": "^1.3.0",
+    "@radix-ui/react-label": "^2.1.0",
     "@radix-ui/react-select": "^2.1.2",
     "@radix-ui/react-slider": "^1.2.1",
     "@radix-ui/react-slot": "^1.1.0",