problem 1 in swift