Update main.c

Updating problem sizes