Working on Swift integration