Updated Suggestion for main.c