Modified to correctly set the subclassed uicollectionview