コードの改善

気になった箇所を変更しました