Update main.c

Adapted to the new errors.