System Design Problem Leetcode Python Easygui