small update to main.c