edits to main.c