if文と変数追加
for文を追加し、内部の処理についてコメントを追加。