Updates to main.c