Added updated generate_data.cpp