Update main.c

added variable b