Fix return codes for main()