Added variables to main()