se adecuo el código a swift un poco mas ordenado