Test for 3 added, simplest code is to add one condition and return a
list with the number if the number is greather than 1
Impreative version
First iterative version