Added tree initialization to main()