Fixed main()