Update main.c

Updating problems sizes.