System Design Problem Leetcode Python Coding