Update to work with CocoaPods