Why Integer here? Is it because of the type mismatch above (which should be fixed if you use the code I wrote)?