Add get_data function