#3

Made a few changes in main.c