Validate vector indeces

1 files changed, 5 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index ce548c4..68ff3cb 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -19,7 +19,11 @@ impl Choice {
         let words: Vec<&str> = line.split_whitespace().collect();
 
         match self {
-            Choice::Field(i) => print!("{} ", words[*i as usize]),
+            Choice::Field(i) => {
+                if *i < words.len().try_into().unwrap() {
+                    print!("{} ", words[*i as usize]);
+                }
+            },
             Choice::FieldRange(r) => match r {
                 (None, None) => print!("{}", words.into_iter().collect::<String>()),
                 (Some(start), None) => print!(