メインソース書き換え

動くように修正しました