Using lists in Python