Update main.c

safer inputs