modify main()