Update main.c

update to version 2