Update main.c

Comment - Alon