Add return code to main()

As trivial as the title and diff look.