COD: change the test function to be called main