If you are going for this feature pls consider it to also work for other mobs with different items. This should not be too hard and then it is implemented in a more structured way.