Codewars | Python | 10