Update generate_input,py