Changed main.c to improve testing