Added more to main.c