Minor main() update.