main関数を修正