Converted the model from Obj-C to Swift