Objective c to swift migration