modify main #2