Silakan berikan kode Python yang Anda butuhkan bantuan. Tentukan apakah Anda memerlukan solusi, optimasi, atau penjelasan. Saya akan meninjau kode tersebut dan memberikan bantuan berdasarkan permintaan Anda.