Moved logic to MosaicViewController