Set all code in main()