Clean up main()