diff --git a/SwiftShellTests/Scripts/print_linenumbers.swift b/SwiftShellTests/Scripts/print_linenumbers.swift
index 48b5f1b..7c37851 100755
--- a/SwiftShellTests/Scripts/print_linenumbers.swift
+++ b/SwiftShellTests/Scripts/print_linenumbers.swift
@@ -2,10 +2,9 @@
 
 import SwiftShell
 
-let result = main.stdin.read().characters.split("\n")
-	.enumerate().map { (linenr,line) in "line " + String(linenr+1) + ": " + String(line) }
+main.stdin.read().characters.split("\n")
+	.enumerate().map { (linenr,line) in "\(linenr+1): " + String(line) }
+	.joinWithSeparator("\n").writeTo(&main.stdout)
 
-//  Swift demands we split this up.
-main.stdout.write(result.joinWithSeparator("\n")) 
-
-// TODO:  use .writeTo(&main.stdout)
+// add a final newline at the end
+print("")