System Design Problem Leetcode Python Basic Programs