added collection class back in