System Design Problem Leetcode Python Interview