メインクラスの修正