cleaned up test code in main.c