Update test_with_data.cpp