System Design Problem Leetcode Python Data