Forgot last update of file main.c

Doh!