Improve Swift example