Clean up main.c