modify Main() method