Corrected the final assignment.

Seemed that the exercise was a bit too vague, so I filled in the
function and just left a "_" for where a single number could be.