edit callback function
add data visualization with 3 countries