First exercice in swift