Updating layer attributions (#10)