change print in main()