Modified main.c to look like original one.