Fixing main()