Changed main.c to how I like it