diff --git a/boilerplate/src/main.rs b/boilerplate/src/main.rs
index d2bced3..f7c61a1 100644
--- a/boilerplate/src/main.rs
+++ b/boilerplate/src/main.rs
@@ -17,11 +17,11 @@ fn main() {
 
 // PROBLEM 1 solution
 
-fn problem1(input: InputIter) -> u64 {
+fn problem1<T: BufRead>(input: Lines<T>) -> u64 {
     0
 }
 
 // PROBLEM 2 solution
-fn problem2(input: InputIter) -> u64 {
+fn problem2<T: BufRead>(input: Lines<T>) -> u64 {
     0
 }
\ No newline at end of file