         
         source.append(" */\n") // TODO: Add input test.md
 
+        // Append sample runner code
+        source.append(
+            "\n" +
+            "//// Don't forget to create a test runner: \n" +
+            "//\n" +
+            "//class \(testClassIdentifier)Runner : \(testClassIdentifier)Runnable {\n" +
+            "//\n" +
+            "//\tfunc run(input: \(inputStructName)) throws -> \(outputStructName) {\n" +
+            "//\t\t//return <\(outputStructName)>\n" +
+            "//\t}\n" +
+            "//}\n" +
+            "\n"
+        )
+
         // Imports
         source.append(
             "import XCTest\n" +

            testOutputs.append("\tlet \(outputVar.name): \(outputVar.type.rawValue)\n")
        }
        
        let outputStructName = "\(testClassIdentifier)Output"
        source.append(
            "struct \(outputStructName): Equatable {\n" +
                testOutputs +

            "}\n"
        )
        
        return source
    }
}
