T-Mobile interview question

Implementing a new feature with Python