Interview tasks are to mimic the real world scenarios in a simpler scale. They are meant to test your approach of a real world scenario.

If you dont think so, just reading the problem, your first solution with lines 17 and 19 removed would solve it.