Collage view controller refectoring