flowkater prob_1 swift solve