Clean-up main.c