Updated test_code