fixing main()