fix layout in main.c