Nice job! Just a small change. And demonstration
of a compact way to do the same in a single line.