How to Fix TypeError: ‘type’ object is not subscriptable in Python Function