Make Class Subscriptable Python