Update main.c

Hunting down seg fault