Updating the src code in the main.c file.