Added assert method to test_main