Added parameters in main()