Tweak main entry point.