added a line in main()