Modify main()