Update extension.swift