Builder Design Pattern in Swift

HouseBuilder.swift

class HouseBuilder {
    var color: UIColor = UIColor.white
    var noOfRooms: Int
    var noOfBathrooms: Int = 1
    var hasBalcony: Bool = false
          
    init(noOfRooms: Int) {
        self.noOfRooms = noOfRooms
    }
          
    func set(color: UIColor) -> HouseBuilder {
        self.color = color
        return self
    }
    func set(noOfBathrooms: Int) -> HouseBuilder {
        self.noOfBathrooms = noOfBathrooms
        return self
    }
    
    func addBalcony() -> HouseBuilder {
        hasBalcony = true
        return self     }
  
    func build() -> House {
        return House(builder: self)
    }
}