UICollectionView cell size 변경 @@ in Swift3 - Xcode 8.2 iOS 10

컬렉션 뷰의 셀 사이즈를 변경 하는데 Swift3.0에선 기존 코드가 적용되지 않습니다.
Swift3.0 에서 셀 사이즈를 변경하는 방법을 알아보았습니다.

- 기존 코드 -

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {

  return CGSize(width: 150, height: 150) 

}

- Swift3 - 

class ViewController: UICollectionViewDelegateFlowLayout, UICollectionViewDelegate { 

}

* UICollectionViewDelegateFlowLayout을 추가해주는 부분이 변경되었습니다.

 func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize { 

  return CGSize(width: 150, height: 150)

} }