edit main()