modify class Main