1 files changed, 6 insertions, 2 deletions

diff --git a/download.tcl b/download.tcl
index 2e64865..31d4c82 100755
--- a/download.tcl
+++ b/download.tcl
@@ -1,15 +1,19 @@
 #!/usr/bin/env expect
 
 set core 0
+set wordsperline 1
 
-if {$::argc > 0} {
+if {$::argc > 1} {
+    set core [lindex $::argv 0]
+    set wordsperline [lindex $::argv 1]
+} elseif {$::argc > 0} {
     set core [lindex $::argv 0]
 }
 
 spawn nios2-terminal --instance 0
 expect -ex "connected to hardware target"
 
-send "@[format %x [expr $core * 32768]]\n"
+send "@[format %x [expr $core * 32768 / $wordsperline]]\n"
 
 while {[gets stdin line] >= 0} {
     send "$line\n"