Refactor of file main.c