refactor
pass selected columns to Heatmap component via callback
fix test - increase timeout