Main class 수정.