Python Make Class Subscriptable