Data structure in Python