add some code to main()
add one line