Add support for Swift