Added a collection decorator to get the item count