Update main.c

final