Modifications to test.c file