Make the code more pythonic

-   use itertools to limit the range of the selections to use
-   use a list comprehension to fetch the selections