Small tweak on calling main.

Nothing really fancy but just better conforming to best practices.