fix -n flag

README.md

@@ -173,6 +173,8 @@ Done:
 
 To (possibly) do:
 
+- [ ] Add flag to repeat input lines (useful when reading from files)
+- [ ] Make -n flag work
 - [ ] Add functions. Now easier to do with an explicit representation of the stakc
 - [ ] Think about how to draw a histogram from samples
 - [ ] Dump samples to file

fermi.go

@@ -411,9 +411,9 @@ replForLoop:
 
 func main() {
 	num_samples_flag := flag.Int("n", N_SAMPLES, "Specifies the number of samples to draw when using samples")
-	N_SAMPLES = *num_samples_flag
 	filename := flag.String("f", "", "Specifies a file with a model to run")
 	flag.Parse()
+	N_SAMPLES = *num_samples_flag
 
 	var reader *bufio.Reader = nil
 	if *filename != "" {