Add get_data method