update to collection class