A simple solution to the first challange of Swift