Swift code optimization