Green test for 2
Refactoring with array
Failing test for 3