Add a callback to select which element are written

Add a callback to select which element are written #239