Updating main.c