List comprehension -- CodingBat exercise