modified class structure
add collection