main関数の修正