Update code to work with Xcode 5 beta