Updated get_data function