pierwsza część kodu działa dobrze, ale nie rozumiem jak poradzić sobie z drugą częścią