diff --git a/exercises/01.advanced-tools/02.problem.structured/src/tools.ts b/exercises/01.advanced-tools/02.problem.structured/src/tools.ts
index 19fb022..54b701c 100644
--- a/exercises/01.advanced-tools/02.problem.structured/src/tools.ts
+++ b/exercises/01.advanced-tools/02.problem.structured/src/tools.ts
@@ -330,7 +330,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 				openWorldHint: false,
 			} satisfies ToolAnnotations,
 			inputSchema: entryTagIdSchema,
-			// 🐨 add an outputSchema here with a tag that is a tagSchema and an entry that is an entrySchema
+			// 🐨 add an outputSchema here with a success boolean and a tagEntry that is an entryTagSchema
 		},
 		async ({ entryId, tagId }) => {
 			const tag = await agent.db.getTag(tagId)
@@ -382,7 +382,7 @@ export async function initializeTools(agent: EpicMeMCP) {
 						'If set to > 0, use mock mode and this is the mock wait time in milliseconds',
 					),
 			},
-			// 🐨 add an outputSchema here with a video that is a videoSchema
+			// 🐨 add an outputSchema here with a video that includes videoUri (you're on your own here!)
 		},
 		async ({ year = new Date().getFullYear(), mockTime }) => {
 			const entries = await agent.db.getEntries()