Changed the code to use iterators instead of manually retrieving
and removing elements from the collection.