Actually setting up the data.
Merge pull request #1 from katfang/porting-demo-to-swift

Demo ported to swift.