i like the idea. please provide me with the entire new updated code based on your suggestions. make sure to include all codes as I am starting with a blank main.py file.  Write your response with exactly 4 paragraphs, separated with the markdown divider: ***.
user