add a changed main()