fixed a test and updated main.c