Instagramの検索画面のようなグリットレイアウトをCollectionViewで実装したい

Question

前提・実現したいこと

以下画像の赤い部分のようにCollectionViewCellを表示したい

Swiftで上記のインスタグラム検索画面のようなレイアウトを実装したいです。
現在の進捗はこんな感じで進んでいます。

できていないことは最初の画像の赤枠部分が１つのセルで表示されてしまいます。
どのように実装したら赤枠のようにセルを２つ表示できるのでしょうか？

よろしくお願いいたします。