Improving the burn function

Now the function returns the cluster sizes per mark type