Custom UICollectionViewFlowLayout. (Swift Version)