编写代码

说明包括我所有的编码偏好：
- 我希望Claude返回完整的代码文件（没有"# rest of the code"）
- 只有在复杂的地方才加注释
- 语言偏好（我现在主要使用Python）