updated collection class to contain more data