fixed main()