Modified the tests, to make it possible to implement all numbers instead of only 0 and 1, as suggested in the README for the exercise.