Fixing warnings in main.c