modify toString() and main()